from robot_llm_tools.tools import move_agent
from robot_llm_tools.tools import features_agent

from robot_llm_tools.tools import chat_history


def main(user_input):
    # user_input = input("请输入命令：")
    # if user_input == "结束对话":
    #     print("好的，本次对话结束，如有其他需要，请对我说“你好”")
    #     break
    # 使用Langchain Agent处理用户输入
    # output = agent.run(input=user_input)
    # chat_history = ["你好，我是一个机器人，我能为你做什么呢？", "你好"]
    output = move_agent.run(
        input=user_input, 
        chat_history=chat_history,
        handle_parsing_errors=True
        )
    return output
    
if __name__ == '__main__':
    while True:
        user_input = input("请输入命令：")
        output = main(user_input)
        print(output)
